package com.xiaoyu.awakening.ai.evaluation.service.impl;

import com.xiaoyu.awakening.ai.evaluation.dto.CreateSingleEvalTaskDTO;
import com.xiaoyu.awakening.ai.evaluation.dto.EvalDataSetGenerateContext;
import com.xiaoyu.awakening.ai.evaluation.service.IAwkEvalDataSetService;
import org.springframework.stereotype.Service;

import java.util.ArrayList;
import java.util.List;

/**
 * 模拟数据集生成服务
 *
 * @author wangxiaoyu
 * @date 2025/8/21 14:58
 * @since v1.0
 */
@Service
public class AwkEvalDataSetMockServiceImpl extends BaseEvalDataSetServiceImpl implements IAwkEvalDataSetService {
    @Override
    public EvalDataSetGenerateContext generate(CreateSingleEvalTaskDTO createSingleEvalTaskDTO) {
        return super.generate(createSingleEvalTaskDTO);
    }

    @Override
    public List<String> doLoadDocuments(CreateSingleEvalTaskDTO createSingleEvalTaskDTO) {
        List<String> documents = new ArrayList<>();
        documents.add("""
                1、朱自清《荷塘月色》片段
                
                路上只我一个人，背着手踱着。这一片天地好像是我的;我也像超出了平常旳自己，到了另一世界里。我爱热闹，也爱冷静;爱群居，也爱独处。像今晚上，一个人在这苍茫旳月下，什么都可以想，什么都可以不想，便觉是个自由的人。白天里一定要做的事，一定要说的话，现在都可不理。这是独处的妙处，我且受用这无边的荷香月色好了。
                
                曲曲折折的荷塘上面，弥望旳是田田的叶子。叶子出水很高，像亭亭旳舞女旳裙。层层的叶子中间，零星地点缀着些白花，有袅娜(niǎo,nuó)地开着旳，有羞涩地打着朵儿旳;正如一粒粒的明珠，又如碧天里的星星，又如刚出浴的美人。微风过处，送来缕缕清香，仿佛远处高楼上渺茫的歌声似的。这时候叶子与花也有一丝的颤动，像闪电般，霎时传过荷塘的那边去了。叶子本是肩并肩密密地挨着，这便宛然有了一道凝碧的波痕。叶子底下是脉脉(mò)的流水，遮住了，不能见一些颜色;而叶子却更见风致了。
                
                月光如流水一般，静静地泻在这一片叶子和花上。薄薄的青雾浮起在荷塘里。叶子和花仿佛在牛乳中洗过一样;又像笼着轻纱的梦。虽然是满月，天上却有一层淡淡的云，所以不能朗照;但我以为这恰是到了好处——酣眠固不可少，小睡也别有风味的。月光是隔了树照过来的，高处丛生的灌木，落下参差的斑驳的黑影，峭楞楞如鬼一般;弯弯的杨柳的稀疏的倩影，却又像是画在荷叶上。塘中的月色并不均匀;但光与影有着和谐的旋律，如梵婀(ē)玲(英语violin小提琴的译音)上奏着的名曲。
                
                荷塘的四面，远远近近，高高低低都是树，而杨柳最多。这些树将一片荷塘重重围住;只在小路一旁，漏着几段空隙，像是特为月光留下的。树色一例是阴阴的，乍看像一团烟雾;但杨柳的丰姿，便在烟雾里也辨得出。树梢上隐隐约约的是一带远山，只有些大意罢了。树缝里也漏着一两点路灯光，没精打采的，是渴睡人的眼。这时候最热闹的，要数树上的蝉声与水里的蛙声;但热闹是它们的，我什么也没有。""");
        documents.add("""
                2、鲁迅《从百草园到三味书屋》片段
                
                不必说碧绿的菜畦，光滑的石井栏，高大的皂荚树，紫红的桑椹;也不必说鸣蝉在树叶里长吟，肥胖的黄蜂伏在菜花上，轻捷的叫天子(云雀)忽然从草间直窜向云霄里去了。单是周围的短短的泥墙根一带，就有无限趣味。油蛉在这里低唱，蟋蟀们在这里弹琴。翻开断砖来，有时会遇见蜈蚣;还有斑蝥，倘若用手指按住它的脊梁，便会拍的一声，从后窍喷出一阵烟雾。何首乌藤和木莲藤缠络着，木莲有莲房一般的果实，何首乌有拥肿的根。有人说，何首乌根是有象人形的，吃了便可以成仙，我于是常常拔它起来，牵连不断地拔起来，也曾因此弄坏了泥墙，却从来没有见过有一块根象人样。如果不怕刺，还可以摘到覆盆子，象小珊瑚珠攒成的小球，又酸又甜，色味都比桑椹要好得远。""");
        return documents;
    }

    @Override
    protected EvalDataSetGenerateContext.DataSetSample doRagFlow(String userInput, CreateSingleEvalTaskDTO createSingleEvalTaskDTO) {
        EvalDataSetGenerateContext.DataSetSample dataSetSample = new EvalDataSetGenerateContext.DataSetSample();
        dataSetSample.setUserInput(userInput);
        dataSetSample.setResponse("success");
        dataSetSample.setRetrievedContexts(List.of("document1", "document2"));
        return dataSetSample;
    }
}
